iT邦幫忙

2022 iThome 鐵人賽
DAY 15
0
自我挑戰組

AI ：PyTorch系列 第 15 篇

第十五天 構建模型

14th鐵人賽
350 瀏覽

  • 分享至 

  • xImage
    •  

前幾天我有學習到word embedding的定義
今天就來手寫一個embedding的class
先新建一個py文件
來構建模型
創建類

https://ithelp.ithome.com.tw/upload/images/20220929/20152487cQMIbKYgoE.png

添加forword的函數

https://ithelp.ithome.com.tw/upload/images/20220929/20152487xLzsOKjyYM.png

最後添加損失函數看看訓練效果

https://ithelp.ithome.com.tw/upload/images/20220929/20152487kN84BTlVXo.png

但是這樣子的程式來測試文本情感分類是會報錯的
原因是我們之前準備數據時候講內容存入dataset時候不應該是list 應該換成longtensor
在dataloader函數裡面將他改為longtensor就好了


上一篇
第十四天 文本序列化
下一篇
第十六天 循環神經網絡
系列文
AI ：PyTorch 共 30 篇
  1. 26
    第二十六天 聊天機器人分詞訓練模型的封裝
  2. 27
    第二十七天 哈夫曼樹
  3. 28
    第二十八天 seq2seq模型
  4. 29
    第二十九天 attention
  5. 30
    第三十天 总结
完整目錄
圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

尚未有邦友留言

立即登入留言
iThome鐵人賽
參賽組數
1064 組
團體組數
40 組
累計文章數
22211 篇
完賽人數
602 人
iT+看影片追技術 看影片追技術 看更多
熱門tag 看更多
15th鐵人賽 16th鐵人賽 13th鐵人賽 14th鐵人賽 12th鐵人賽 11th鐵人賽 鐵人賽 2019鐵人賽 javascript 2018鐵人賽 python 2017鐵人賽 windows php c# windows server linux css react vue.js
熱門問題
熱門回答
熱門文章
IT邦幫忙